import os
import cv2
# 输入原始图像存在的文件夹
datadir = "/home/jiayu/Desktop/Ma/img/"
# 设置保存路径
save_path = '/home/jiayu/Desktop/Ma_0'
if not os.path.exists(save_path):#如果路径不存在
	os.makedirs(save_path)
# 输入你想要resize的图像尺寸。
size = 384
# # 获取原始图像宽高。
# height, width = image.shape[0], image.shape[1]
# # 等比例缩放尺度。
# scale = height/size    # 1
# # 获得相应等比例的图像宽度。
# width_size = int(width/scale)  # 2
#使用os.path生成路径。
path = os.path.join(datadir)
#使用os.listdir返回path路径下所有图像文件。
img_list = os.listdir(path)
# 遍历图像文件
for i in img_list:
    print(i.dtype)
    img_array = cv2.imread(os.path.join(path, i), cv2.IMREAD_COLOR)

    height, width = img_array.shape[0], img_array.shape[1]
    scale = height / size
    width_size = int(width / scale)

    # resize
    new_image = cv2.resize(img_array, (width_size, size))
    save_path = os.path.join(save_path, str(i)) # 保存的图片与原始图片同名
    #保存图片
    cv2.imwrite(save_path, new_image)
